Home
last modified time | relevance | path

Searched defs:replicate (Results 1 – 22 of 22) sorted by relevance

/external/eigen/test/
Dcuda_basic.cu64 struct replicate { struct
74 MapType(out+i*stride+0*step, x1.rows()*2, x1.cols()*2) = x1.replicate(2,2); in operator ()() argument
Darray_replicate.cpp12 template<typename MatrixType> void replicate(const MatrixType& m) in replicate() function
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py40 def replicate(cls): member in Sharding
175 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_variable_runtime_reformatting.cc141 TF::WhileOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
246 tf_device::ReplicateOp AddInputsToReplicateOp(tf_device::ReplicateOp replicate, in AddInputsToReplicateOp()
371 void HandleReplicateOp(TF::WhileOp while_op, tf_device::ReplicateOp replicate, in HandleReplicateOp()
490 tf_device::ReplicateOp replicate; in runOnModule() local
Dtpu_dynamic_padding_mapper.cc58 tf_device::LaunchFuncOp launch_func, tf_device::ReplicateOp replicate) { in GetRemappedReplicatedInputIndices()
171 auto replicate = in RemapAndAssignPaddingMaps() local
Dannotate_parameter_replication.cc60 auto replicate = launch_func.getParentOfType<tf_device::ReplicateOp>(); in runOnModule() local
Dtpu_dynamic_layout_pass.cc145 int64_t replicate_arg_index, tf_device::ReplicateOp replicate, in HandleReplicatedInputs()
Dtpu_rewrite_pass.cc420 tf_device::ReplicateOp replicate = in Rewrite() local
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py140 def replicate(dataset, devices): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_and_filter_fusion_test.py57 replicate = lambda x: (x, x) function
/external/swiftshader/src/Common/
DTypes.hpp143 inline float4 replicate(float f) in replicate() function
/external/eigen/Eigen/src/Core/
DDenseBase.h554 const Replicate<Derived, Dynamic, Dynamic> replicate(Index rowFactor, Index colFactor) const in replicate() function
/external/swiftshader/src/System/
DTypes.hpp199 inline constexpr float4 replicate(float f) in replicate() function
/external/swiftshader/src/Device/
DPixelProcessor.hpp127 static int64_t replicate(int b) in replicate() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.cc178 } else if (auto replicate = llvm::dyn_cast<tf_device::ReplicateOp>(op)) { in AnalyzeFunction() local
/external/swiftshader/src/Renderer/
DPixelProcessor.hpp157 static int64_t replicate(int b) in replicate() function
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py784 def replicate(computation, function
/external/swiftshader/src/Reactor/
DLLVMReactor.cpp2514 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Int4() local
2682 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in UInt4() local
2936 Value *replicate = Nucleus::createShuffleVector(insert, insert, swizzle); in Float4() local
DSubzeroReactor.cpp3392 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Int4() local
3629 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in UInt4() local
3818 Value *replicate = Nucleus::createShuffleVector(vector, vector, swizzle); in Float4() local
/external/vixl/src/aarch32/
Dmacro-assembler-aarch32.cc1301 static inline RES replicate(T imm) { in replicate() function
/external/pcre/dist2/src/
Dpcre2_compile.c6866 int replicate = repeat_min; in compile_branch() local
/external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmTypeTests.cpp2500 std::string SpvAsmTypeTests<T>::replicate (const std::string& replicant, in replicate() function in vkt::SpirVAssembly::SpvAsmTypeTests