| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_variable_runtime_reformatting.cc | 99 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
224 tf_device::ReplicateOp replicate, in AddInputsToReplicateOp()
336 tf_device::ReplicateOp replicate) { in HandleReplicateOp()
478 tf_device::ReplicateOp replicate; in runOnOperation() local
|
| H A D | tpu_colocate_composite_resource_ops.cc | 60 tf_device::ReplicateOp replicate) { in GetResourceOpsUsingCompositeArgsInReplicate()
122 auto replicate = execute_launch->getParentOfType<tf_device::ReplicateOp>(); in runOnOperation() local
|
| H A D | annotate_parameter_replication.cc | 63 auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); in runOnOperation() local
|
| H A D | tpu_rewrite_pass.cc | 402 tf_device::ReplicateOp replicate, in AssignDevicesToReplicate()
720 tf_device::ReplicateOp replicate = in Rewrite() local
|
| H A D | tpu_dynamic_layout_pass.cc | 168 mlir::BlockArgument replicate_arg, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/parallel/ |
| H A D | data_parallel.h | 149 std::vector<std::shared_ptr<ModuleType>> replicate( in replicate() function
168 std::vector<ModuleHolder<ModuleType>> replicate( in replicate() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| H A D | xla_sharding.py | 36 def replicate(cls): member in Sharding
308 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): function
|
| /aosp_15_r20/external/intel-media-driver/media_driver/agnostic/gen11/codec/kernel_free/Source/ |
| H A D | downscale_kernel_genx.cpp | 131 vector<ushort, 1> replicate; in hme_frame_downscale() local
404 vector<ushort, 1> replicate; in hme_frame_downscale2() local
|
| H A D | downscale_convert_kernel_genx.cpp | 107 vector<ushort, 1> replicate; in DS_Convert() local
|
| /aosp_15_r20/external/eigen/test/ |
| H A D | gpu_basic.cu | 219 struct replicate { struct
229 MapType(out+i*stride+0*step, x1.rows()*2, x1.cols()*2) = x1.replicate(2,2); in operator ()() argument
|
| H A D | array_replicate.cpp | 12 template<typename MatrixType> void replicate(const MatrixType& m) in replicate() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | PadNd.h | 9 replicate, enumerator
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | replicate.py | 115 def replicate( function
|
| H A D | data_parallel.py | 196 def replicate( member in DataParallel
|
| /aosp_15_r20/external/bazelbuild-rules_go/go/tools/builders/ |
| H A D | replicate.go | 149 func replicate(src, dst string, options ...replicateOption) error { func
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/experimental/ |
| H A D | test_local_map.py | 28 replicate = [Replicate()] # replicate placements on 1-d mesh variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| H A D | map_and_filter_fusion_test.py | 54 replicate = lambda x: (x, x) function
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/ |
| H A D | replicate.py | 179 def replicate( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | distribute.py | 331 def replicate(dataset, devices): function
|
| /aosp_15_r20/tools/tradefederation/core/src/com/android/tradefed/command/ |
| D | ICommandOptions.java | 229 public void setReplicateSetup(boolean replicate); in setReplicateSetup()
|
| /aosp_15_r20/external/eigen/Eigen/src/Core/ |
| H A D | DenseBase.h | 600 const Replicate<Derived, Dynamic, Dynamic> replicate(Index rowFactor, Index colFactor) const in replicate() function
|
| /aosp_15_r20/external/swiftshader/src/System/ |
| H A D | Types.hpp | 232 inline constexpr float4 replicate(float f) in replicate() function
|
| /aosp_15_r20/external/swiftshader/src/Reactor/ |
| H A D | LLVMReactor.cpp | 2567 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int4() local
2771 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt4() local
3057 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float4() local
4160 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int() local
4300 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt() local
4395 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float() local
|
| H A D | SubzeroReactor.cpp | 3295 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Int4() local
3549 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in UInt4() local
3713 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Float4() local
4742 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Int() local
4936 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in UInt() local
5059 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Float() local
|
| /aosp_15_r20/external/swiftshader/src/Device/ |
| H A D | PixelProcessor.hpp | 141 static int64_t replicate(int b) in replicate() function
|