| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_colocate_composite_resource_ops.cc | 65 tf_device::ReplicateOp replicate) { in GetResourceOpsUsingCompositeArgsInReplicate()
127 auto replicate = execute_launch->getParentOfType<tf_device::ReplicateOp>(); in runOnFunction() local
|
| D | tpu_variable_runtime_reformatting.cc | 155 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
280 tf_device::ReplicateOp replicate, in AddInputsToReplicateOp()
392 tf_device::ReplicateOp replicate) { in HandleReplicateOp()
534 tf_device::ReplicateOp replicate; in runOnOperation() local
|
| D | annotate_parameter_replication.cc | 62 auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); in runOnOperation() local
|
| D | tpu_rewrite_pass.cc | 397 tf_device::ReplicateOp replicate, in AssignDevicesToReplicate()
559 tf_device::ReplicateOp replicate = in Rewrite() local
|
| D | tpu_dynamic_layout_pass.cc | 192 mlir::BlockArgument replicate_arg, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
|
| D | tpu_space_to_depth_pass.cc | 408 tf_device::ReplicateOp replicate, in HandleHostReplicatedInputs()
|
| /external/eigen/test/ |
| D | array_replicate.cpp | 12 template<typename MatrixType> void replicate(const MatrixType& m) in replicate() function
|
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 40 def replicate(cls): member in Sharding
261 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_and_filter_fusion_test.py | 58 replicate = lambda x: (x, x) function
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 316 def replicate(dataset, devices): function
|
| /external/eigen/Eigen/src/Core/ |
| D | DenseBase.h | 600 const Replicate<Derived, Dynamic, Dynamic> replicate(Index rowFactor, Index colFactor) const in replicate() function
|
| /external/swiftshader/src/System/ |
| D | Types.hpp | 232 inline constexpr float4 replicate(float f) in replicate() function
|
| /external/swiftshader/src/Device/ |
| D | PixelProcessor.hpp | 137 static int64_t replicate(int b) in replicate() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 500 auto replicate = cluster->getParentOfType<mlir::tf_device::ReplicateOp>(); in GetHostDeviceOutsideComputation() local
|
| D | tpu_rewrite_device_util_test.cc | 794 auto replicate = builder.create<mlir::tf_device::ReplicateOp>( in TEST() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 896 def replicate( function
|
| /external/swiftshader/src/Reactor/ |
| D | LLVMReactor.cpp | 2595 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int4() local
2799 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt4() local
3123 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float4() local
|
| D | SubzeroReactor.cpp | 3480 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Int4() local
3746 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in UInt4() local
3934 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Float4() local
|
| /external/vixl/src/aarch32/ |
| D | macro-assembler-aarch32.cc | 1301 static inline RES replicate(T imm) { in replicate() function
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 4349 string replicate; in LowerOutsideCompilationFunctionalNodes() local
4797 string replicate; in RewriteTPUReplicateNode() local
|
| /external/pcre/src/ |
| D | pcre2_compile.c | 7108 int replicate = repeat_min; in compile_branch() local
|
| /external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/ |
| D | vktSpvAsmTypeTests.cpp | 2504 std::string SpvAsmTypeTests<T>::replicate (const std::string& replicant, in replicate() function in vkt::SpirVAssembly::SpvAsmTypeTests
|