Home
last modified time | relevance | path

Searched defs:init_value (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc84 auto* init_value = builder.AddInstruction( in TEST_F() local
143 auto* init_value = builder.AddInstruction( in TEST_F() local
197 auto* init_value = builder.AddInstruction( in TEST_F() local
237 auto* init_value = builder.AddInstruction( in TEST_F() local
280 auto* init_value = builder.AddInstruction( in TEST_F() local
333 auto* init_value = builder.AddInstruction( in TEST_F() local
382 auto* init_value = builder.AddInstruction( in TEST_F() local
431 auto* init_value = builder.AddInstruction( in TEST_F() local
459 auto* init_value = builder.AddInstruction( in TEST_F() local
Dwhile_loop_constant_sinking.cc54 const HloInstruction& init_value = *while_instr->operand(0); in TrySinkingConstantsIntoWhileLoop() local
Dcall_inliner_test.cc102 HloInstruction* init_value = outer.AddInstruction( in TEST_F() local
Dhlo_creation_utils.cc323 HloInstruction* init_value, in MakeReduceHlo()
350 HloInstruction* init_value, in MakeReduceHlo()
/external/OpenCL-CTS/test_conformance/spirv_new/
Dtest_op_spec_constant.cpp140 cl_uchar init_value = value; variable
152 cl_uchar init_value = value; variable
/external/tensorflow/tensorflow/python/autograph/utils/
Dcompat_util.py30 def __init__(self, init_value): argument
/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op_test.cc145 std::vector<float> init_value(num_slices, 0.0f); in TEST_P() local
196 std::vector<float> init_value(num_slices, 0.0f); in TEST_P() local
250 std::vector<float> init_value(num_slices, 0.0f); in TEST_P() local
333 std::vector<float> init_value(num_slices, 0.0f); in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h450 void ShapeTree<T>::InitChildren(const Shape& shape, const T& init_value, in InitChildren()
559 ShapeTree<T>::ShapeTree(Shape shape, const T& init_value) in ShapeTree()
572 ShapeTree<T>::ShapeTree(const Shape* shape, const T& init_value) in ShapeTree()
585 const T& init_value) in ShapeTree()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc73 XlaOp ComputeSums(XlaOp operand, XlaOp init_value, in ComputeSums()
142 auto init_value = MinValue(b, dtype); in MaxPool() local
157 auto init_value = Zero(b, dtype); in AvgPool() local
/external/tensorflow/tensorflow/python/training/
Devaluation_test.py51 def local_variable(init_value, name): argument
/external/pcre/src/sljit/
DsljitNativePPC_32.c261 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 reg, sljit_sw init_value) in emit_const()
DsljitNativeSPARC_32.c260 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const()
DsljitNativePPC_64.c482 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 reg, sljit_sw init_value) in emit_const()
DsljitNativeMIPS_32.c418 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const()
DsljitNativeMIPS_64.c536 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc255 SmallVector<Value, 1> init_value = {rewriter->create<memref::LoadOp>( in CreateReduceOpInNestedParallelLoops() local
393 Value init_value = rewriter->create<memref::LoadOp>( in CreateParallelLoopsToTraverseOutputAndWindow() local
527 Value init_value = b->create<memref::LoadOp>(loc, s_and_s_op.init_value()); in InitializeOutput() local
Dlhlo_legalize_to_gpu.cc101 auto init_value = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc111 const auto init_value = in XLA_TEST_P() local
647 auto init_value = in DoIt() local
1049 auto init_value = in XLA_TEST_P() local
1263 auto init_value = in DoIt() local
1463 auto init_value = in XLA_TEST_P() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc394 xla::XlaOp init_value = XlaHelpers::Zero(ctx->builder(), input_type(2)); in Compile() local
637 auto init_value = xla::MinValue(b, xla::F32); in Compile() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreduce.h158 const T init_value, T* data) { in InitTensorDataForReduce()
183 T init_value, in ReduceGeneric()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dloop_descriptor.cpp723 int64_t init_value = 0; in FindNumberOfIterations() local
756 int64_t init_value, int64_t step_value) const { in GetIterations()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_descriptor.cpp723 int64_t init_value = 0; in FindNumberOfIterations() local
756 int64_t init_value, int64_t step_value) const { in GetIterations()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_descriptor.cpp723 int64_t init_value = 0; in FindNumberOfIterations() local
756 int64_t init_value, int64_t step_value) const { in GetIterations()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2452 XlaOp XlaBuilder::Reduce(XlaOp operand, XlaOp init_value, in Reduce()
2505 XlaOp XlaBuilder::ReduceAll(XlaOp operand, XlaOp init_value, in ReduceAll()
2515 XlaOp XlaBuilder::ReduceWindow(XlaOp operand, XlaOp init_value, in ReduceWindow()
2589 const auto& init_value = init_values[0]; in ReduceWindowWithGeneralPadding() local
2633 const auto& init_value = init_values[i]; in ReduceWindowInternal() local
2658 const Shape& shape, XlaOp operand, XlaOp init_value, in ReduceWindowInternal()
3119 XlaOp init_value, in SelectAndScatter()
3165 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterInternal()
3195 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
4434 XlaOp Reduce(const XlaOp operand, const XlaOp init_value, in Reduce()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util_test.cc258 Status MakeLoop(Scope s, Output init_value, absl::string_view loop_name) { in MakeLoop()

123