| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| 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
|
| 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
|
| D | annotate_parameter_replication.cc | 63 auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); in runOnOperation() local
|
| D | tpu_dynamic_layout_pass.cc | 168 mlir::BlockArgument replicate_arg, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
|
| D | tpu_rewrite_pass.cc | 402 tf_device::ReplicateOp replicate, in AssignDevicesToReplicate()
720 tf_device::ReplicateOp replicate = in Rewrite() local
|
| D | tpu_space_to_depth_pass.cc | 409 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 | 36 def replicate(cls): member in Sharding
308 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 | 54 replicate = lambda x: (x, x) function
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 331 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 | 141 static int64_t replicate(int b) in replicate() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 496 auto replicate = cluster->getParentOfType<mlir::tf_device::ReplicateOp>(); in GetHostDeviceOutsideComputation() local
|
| D | tpu_rewrite_device_util_test.cc | 817 auto replicate = builder.create<mlir::tf_device::ReplicateOp>( in TEST() local
|
| /external/swiftshader/src/Reactor/ |
| D | LLVMReactor.cpp | 2574 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int4() local
2778 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt4() local
3064 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float4() local
4167 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int() local
4307 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt() local
4402 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float() local
|
| 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
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 915 def replicate( function
|
| /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 | 4296 string replicate; in LowerOutsideCompilationFunctionalNodes() local
4740 string replicate; in RewriteTPUReplicateNode() local
|
| /external/pcre/src/ |
| D | pcre2_compile.c | 7110 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
|