| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_colocate_composite_resource_ops.cc | 57 tf_device::ReplicateOp replicate) { in GetResourceOpsUsingCompositeArgsInReplicate()
119 auto replicate = execute_launch->getParentOfType<tf_device::ReplicateOp>(); in runOnFunction() local
|
| D | tpu_variable_runtime_reformatting.cc | 146 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
271 tf_device::ReplicateOp replicate, in AddInputsToReplicateOp()
383 tf_device::ReplicateOp replicate) { in HandleReplicateOp()
525 tf_device::ReplicateOp replicate; in runOnOperation() local
|
| D | tpu_dynamic_padding_mapper.cc | 59 tf_device::ClusterFuncOp cluster_func, tf_device::ReplicateOp replicate, in GetRemappedReplicatedInputIndices()
183 auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); in RemapAndAssignPaddingMaps() local
|
| D | annotate_parameter_replication.cc | 61 auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); in runOnOperation() local
|
| D | tpu_rewrite_pass.cc | 412 tf_device::ReplicateOp replicate, in AssignDevicesToReplicate()
572 tf_device::ReplicateOp replicate = in Rewrite() local
|
| D | tpu_dynamic_layout_pass.cc | 185 mlir::BlockArgument replicate_arg, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
|
| D | tpu_space_to_depth_pass.cc | 454 tf_device::ReplicateOp replicate, in HandleHostReplicatedInputs()
|
| /external/eigen/test/ |
| D | cuda_basic.cu | 64 struct replicate { struct
74 MapType(out+i*stride+0*step, x1.rows()*2, x1.cols()*2) = x1.replicate(2,2); in operator ()() argument
|
| D | array_replicate.cpp | 12 template<typename MatrixType> void replicate(const MatrixType& m) in replicate() function
|
| /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 | 57 replicate = lambda x: (x, x) function
|
| /external/swiftshader/src/Common/ |
| D | Types.hpp | 143 inline float4 replicate(float f) in replicate() function
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 311 def replicate(dataset, devices): function
|
| /external/eigen/Eigen/src/Core/ |
| D | DenseBase.h | 554 const Replicate<Derived, Dynamic, Dynamic> replicate(Index rowFactor, Index colFactor) const in replicate() function
|
| /external/swiftshader/src/System/ |
| D | Types.hpp | 221 inline constexpr float4 replicate(float f) in replicate() function
|
| /external/swiftshader/src/Device/ |
| D | PixelProcessor.hpp | 131 static int64_t replicate(int b) in replicate() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util_test.cc | 762 auto replicate = builder.create<mlir::tf_device::ReplicateOp>( in TEST() local
|
| D | tpu_rewrite_device_util.cc | 492 auto replicate = cluster->getParentOfType<mlir::tf_device::ReplicateOp>(); in GetHostDeviceOutsideComputation() local
|
| /external/swiftshader/src/Renderer/ |
| D | PixelProcessor.hpp | 157 static int64_t replicate(int b) in replicate() function
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 881 def replicate( function
|
| /external/swiftshader/src/Reactor/ |
| D | LLVMReactor.cpp | 2627 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int4() local
2810 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt4() local
3134 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float4() local
|
| D | SubzeroReactor.cpp | 3479 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Int4() local
3738 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in UInt4() local
3926 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Float4() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3842 string replicate; in LowerOutsideCompilationFunctionalNodes() local
4238 string replicate; in RewriteTPUReplicateNode() local
|
| /external/vixl/src/aarch32/ |
| D | macro-assembler-aarch32.cc | 1301 static inline RES replicate(T imm) { in replicate() function
|
| /external/pcre/dist2/src/ |
| D | pcre2_compile.c | 6994 int replicate = repeat_min; in compile_branch() local
|