Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
H A Dhlo_creation_utils.cc385 HloInstruction* init_value, in MakeReduceHlo()
399 HloInstruction* init_value, in MakeReduceHlo()
421 HloInstruction* init_value, in MakeReduceHlo()
H A Dwhile_loop_constant_sinking.cc72 const HloInstruction& init_value = *while_instr->operand(0); in TrySinkingConstantsIntoWhileLoop() local
H A Dhlo_evaluator_test.cc188 auto init_value = b.AddInstruction( in ReduceWindowMaxIotaTest() local
2562 HloInstruction* init_value = b.AddInstruction( in TEST_F() local
2597 auto init_value = b.AddInstruction( in BM_ReducePrecisely() local
2639 auto init_value = b.AddInstruction( in TEST_P() local
2681 auto init_value = b.AddInstruction( in TEST_P() local
2798 auto init_value = b.AddInstruction( in TEST_P() local
2851 auto init_value = b.AddInstruction( in TEST_P() local
H A Dselect_and_scatter_expander.cc38 auto* init_value = sas->mutable_operand(2); in ExpandInstruction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_tree.h116 ShapeTree(Shape shape, const T& init_value) in ShapeTree()
119 ShapeTree(const Shape* shape, const T& init_value) in ShapeTree()
317 ShapeTree(std::shared_ptr<Shape> shape, const T& init_value) in ShapeTree()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A 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
/aosp_15_r20/external/OpenCL-CTS/test_conformance/spirv_new/
H A Dtest_op_spec_constant.cpp140 cl_uchar init_value = value; variable
152 cl_uchar init_value = value; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DReduceUtils.h45 inline vec_scalar_t<scalar_t> init_value() { in init_value() function
63 inline vec_scalar_t<scalar_t> init_value(const std::optional<Scalar>& initial) { in init_value() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_window_test.cc109 const auto init_value = in XLA_TEST_P() local
645 auto init_value = in DoIt() local
1074 auto init_value = in XLA_TEST_P() local
1312 auto init_value = in DoIt() local
1536 auto init_value = in XLA_TEST_P() local
/aosp_15_r20/external/mesa3d/src/compiler/nir/tests/
H A Dloop_analyze_tests.cpp54 uint32_t init_value; member
114 uint32_t init_value; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2579 XlaOp XlaBuilder::Reduce(XlaOp operand, XlaOp init_value, in Reduce()
2632 XlaOp XlaBuilder::ReduceAll(XlaOp operand, XlaOp init_value, in ReduceAll()
2642 XlaOp XlaBuilder::ReduceWindow(XlaOp operand, XlaOp init_value, in ReduceWindow()
2715 const auto& init_value = init_values[0]; in ReduceWindowWithGeneralPadding() local
2759 const auto& init_value = init_values[i]; in ReduceWindowInternal() local
2784 const Shape& shape, XlaOp operand, XlaOp init_value, in ReduceWindowInternal()
3260 XlaOp init_value, in SelectAndScatter()
3306 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterInternal()
3336 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
4583 XlaOp Reduce(const XlaOp operand, const XlaOp init_value, in Reduce()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc403 xla::XlaOp init_value = XlaHelpers::Zero(ctx->builder(), input_type(2)); in Compile() local
648 auto init_value = xla::MinValue(b, xla::F32); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreduce.h160 const T init_value, T* data) { in InitTensorDataForReduce()
185 T init_value, in ReduceGeneric()
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_descriptor.cpp717 int64_t init_value = 0; in FindNumberOfIterations() local
750 int64_t init_value, int64_t step_value) const { in GetIterations()
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/opt/
H A Dloop_descriptor.cpp717 int64_t init_value = 0; in FindNumberOfIterations() local
750 int64_t init_value, int64_t step_value) const { in GetIterations()
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/opt/
H A Dloop_descriptor.cpp717 int64_t init_value = 0; in FindNumberOfIterations() local
750 int64_t init_value, int64_t step_value) const { in GetIterations()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Devaluation_test.py47 def local_variable(init_value, name): argument
/aosp_15_r20/bionic/libc/malloc_debug/
H A DGuardData.cpp42 GuardData::GuardData(DebugData* debug_data, int init_value, size_t num_bytes) in GuardData()
/aosp_15_r20/external/vboot_reference/tests/
H A Dcbfstool_tests.c123 char init_value[] = "INIT_VALUE"; in cbfstool_get_config_string_tests() local
/aosp_15_r20/external/pytorch/tools/stats/
H A Dupload_test_stats.py181 def init_value(test_case: dict[str, Any]) -> dict[str, Any]: function
/aosp_15_r20/external/pcre/src/sljit/
H A DsljitNativeRISCV_32.c115 …jit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value, sljit_ins … in emit_const()
H A DsljitNativeRISCV_64.c168 …jit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value, sljit_ins … in emit_const()
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dmicrobatch.py40 def __init__(self, init_value, reduce_fn): argument

123