Home
last modified time | relevance | path

Searched defs:init_value (Results 1 – 25 of 60) 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()
348 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/dist2/src/sljit/
DsljitNativePPC_32.c252 …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.c468 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 reg, sljit_sw init_value) in emit_const()
DsljitNativeMIPS_32.c419 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const()
DsljitNativeMIPS_64.c515 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const()
/external/adhd/cras/src/server/
Dcras_dsp_ini.h40 float init_value; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc252 SmallVector<Value, 1> init_value = { in CreateReduceOpInNestedParallelLoops() local
387 Value init_value = in CreateParallelLoopsToTraverseOutputAndWindow() local
520 Value init_value = b->create<LoadOp>(loc, s_and_s_op.init_value()); in InitializeOutput() local
Dlhlo_legalize_to_gpu.cc99 auto init_value = rewriter.create<mlir::LoadOp>(loc, std::get<0>(pair)); 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/lite/kernels/internal/reference/
Dreduce.h109 const T init_value, T* data) { in InitTensorDataForReduce()
133 T init_value, in ReduceGeneric()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc355 xla::XlaOp init_value = XlaHelpers::Zero(ctx->builder(), input_type(2)); in Compile() local
596 auto init_value = xla::MinValue(b, xla::F32); in Compile() local
/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/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/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/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2424 XlaOp XlaBuilder::Reduce(XlaOp operand, XlaOp init_value, in Reduce()
2477 XlaOp XlaBuilder::ReduceAll(XlaOp operand, XlaOp init_value, in ReduceAll()
2487 XlaOp XlaBuilder::ReduceWindow(XlaOp operand, XlaOp init_value, in ReduceWindow()
2561 const auto& init_value = init_values[0]; in ReduceWindowWithGeneralPadding() local
2605 const auto& init_value = init_values[i]; in ReduceWindowInternal() local
2630 const Shape& shape, XlaOp operand, XlaOp init_value, in ReduceWindowInternal()
3030 XlaOp init_value, in SelectAndScatter()
3076 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterInternal()
3106 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
4597 XlaOp Reduce(const XlaOp operand, const XlaOp init_value, in Reduce()
[all …]

123