Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc82 auto* init_value = builder.AddInstruction( in TEST_F() local
86 while_body, init_value)); in TEST_F()
141 auto* init_value = builder.AddInstruction( in TEST_F() local
145 while_body, init_value)); in TEST_F()
195 auto* init_value = builder.AddInstruction( in TEST_F() local
199 while_body, init_value)); in TEST_F()
234 auto* init_value = builder.AddInstruction( in TEST_F() local
238 while_body, init_value)); in TEST_F()
268 auto* init_value = builder.AddInstruction( in TEST_F() local
272 while_body, init_value)); in TEST_F()
[all …]
Duser_computation.cc480 TF_ASSIGN_OR_RETURN(const OperationRequest* init_value, in AddReduceInstruction()
481 LookUpRequest(reduce_request.init_value())); in AddReduceInstruction()
492 operand->output_shape(), init_value->output_shape(), in AddReduceInstruction()
648 TF_ASSIGN_OR_RETURN(const OperationRequest* init_value, in AddReduceWindowInstruction()
649 LookUpRequest(reduce_window_request.init_value())); in AddReduceWindowInstruction()
660 operand->output_shape(), init_value->output_shape(), in AddReduceWindowInstruction()
689 TF_ASSIGN_OR_RETURN(const OperationRequest* init_value, in AddSelectAndScatterInstruction()
690 LookUpRequest(select_and_scatter_request.init_value())); in AddSelectAndScatterInstruction()
706 init_value->output_shape(), *scatter_program_shape)); in AddSelectAndScatterInstruction()
1825 PureFunctionalVisitor(session_computation, reduce_request.init_value(), in PureFunctionalVisitor()
[all …]
Dcall_inliner_test.cc101 HloInstruction* init_value = outer.AddInstruction( in TEST_F() local
104 HloInstruction::CreateWhile(pred, call_false, call_false, init_value)); in TEST_F()
Dshape_inference.h134 const Shape& arg, const Shape& init_value,
141 const Shape& operand_shape, const Shape& init_value, const Window& window,
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h115 ShapeTree(Shape shape, const T& init_value);
116 ShapeTree(const Shape* shape, const T& init_value);
289 void InitChildren(const Shape& shape, const T& init_value, Node* node);
471 void ShapeTree<T>::InitChildren(const Shape& shape, const T& init_value, in InitChildren() argument
475 node->children.emplace_back(new Node(init_value)); in InitChildren()
476 InitChildren(shape.tuple_shapes(i), init_value, in InitChildren()
509 ShapeTree<T>::ShapeTree(Shape shape, const T& init_value) in ShapeTree() argument
510 : root_(init_value), in ShapeTree()
516 InitChildren(*shape_, init_value, &root_); in ShapeTree()
520 ShapeTree<T>::ShapeTree(const Shape* shape, const T& init_value) in ShapeTree() argument
[all …]
/external/adhd/cras/src/server/
Dcras_dsp_ini.c124 p->init_value = 0; in parse_ports()
127 float init_value = strtof(str, &endptr); in parse_ports() local
135 p->init_value = init_value; in parse_ports()
206 p->init_value = 0; in add_audio_port()
430 port->init_value); in cras_dsp_ini_dump()
Dcras_dsp_ini.h40 float init_value; member
/external/v8/src/compiler/
Dloop-variable-optimizer.h23 Node* init_value() const { return init_value_; } in init_value() function
42 InductionVariable(Node* phi, Node* arith, Node* increment, Node* init_value, in InductionVariable() argument
47 init_value_(init_value), in InductionVariable()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py1048 init_value=c.Constant(NumpyArrayF32(1)),
1060 init_value=c.Constant(NumpyArrayF64(1)),
1068 init_value=c.ConstantF32Scalar(0),
1077 init_value=c.ConstantF64Scalar(0),
1087 init_value=c.ConstantF32Scalar(0),
1097 init_value=c.ConstantF64Scalar(0),
1107 init_value=c.ConstantF32Scalar(0),
1117 init_value=c.ConstantF64Scalar(0),
1129 init_value=c.ConstantF32Scalar(0),
1149 init_value=c.ConstantF64Scalar(0),
[all …]
Dlocal_computation_builder.cc397 const ComputationDataHandle& init_value, const LocalComputation& scatter) { in SelectAndScatterWithGeneralPadding() argument
400 source, init_value, scatter.computation()); in SelectAndScatterWithGeneralPadding()
470 const ComputationDataHandle& init_value, in Reduce() argument
473 return builder_.Reduce(operand, init_value, local_computation.computation(), in Reduce()
479 const ComputationDataHandle& init_value, in ReduceWindowWithGeneralPadding() argument
485 operand, init_value, local_computation.computation(), window_dimensions, in ReduceWindowWithGeneralPadding()
Dlocal_computation_builder.h192 const ComputationDataHandle& init_value, const LocalComputation& scatter);
237 const ComputationDataHandle& init_value,
243 const ComputationDataHandle& init_value,
Dxla_client.py714 padding, source, init_value, scatter): argument
740 _unwrap_data_handle(source), _unwrap_data_handle(init_value),
874 def Reduce(self, operand, init_value, computation_to_apply, dimensions): argument
889 _unwrap_data_handle(init_value),
893 def ReduceWindow(self, operand, init_value, computation_to_apply, argument
914 _unwrap_data_handle(init_value),
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc39 xla::ComputationDataHandle init_value; in ArgMinMax() local
42 init_value = XlaHelpers::MaxValue(builder, input_type); in ArgMinMax()
45 init_value = XlaHelpers::MinValue(builder, input_type); in ArgMinMax()
53 input, init_value, *reducer, /*dimensions_to_reduce=*/{axis}); in ArgMinMax()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc106 const auto init_value = in TEST_P() local
109 builder_.ReduceWindow(input, init_value, in TEST_P()
591 auto init_value = in DoIt() local
599 /*init_value=*/init_value, in DoIt()
937 auto init_value = in TEST_P() local
940 /*init_value=*/init_value, in TEST_P()
1042 auto init_value = in DoIt() local
1045 /*init_value=*/init_value, in DoIt()
1233 auto init_value = in TEST_P() local
1237 /*init_value=*/init_value, in TEST_P()
Dreduce_test.cc119 ComputationDataHandle init_value; in RunR1ToR0PredTest() local
122 init_value = builder.ConstantR0<bool>(true); in RunR1ToR0PredTest()
125 init_value = builder.ConstantR0<bool>(false); in RunR1ToR0PredTest()
128 builder.Reduce(pred_values, init_value, reduce, in RunR1ToR0PredTest()
157 ComputationDataHandle init_value; in RunR2ToR1PredTest() local
160 init_value = builder.ConstantR0<bool>(true); in RunR2ToR1PredTest()
163 init_value = builder.ConstantR0<bool>(false); in RunR2ToR1PredTest()
167 builder.Reduce(input_pred, init_value, reduce_op, in RunR2ToR1PredTest()
/external/tensorflow/tensorflow/python/training/
Dserver_lib_test.py247 init_value = array_ops.placeholder(dtypes.int32)
248 v = variables.Variable(init_value, validate_shape=False, name="v")
266 sharing_sess_0.run(v.initializer, feed_dict={init_value: 86})
275 sharing_sess_1.run(v.initializer, feed_dict={init_value: [86, 99]})
285 isolate_sess_0.run(v.initializer, feed_dict={init_value: 37})
293 isolate_sess_1.run(v.initializer, feed_dict={init_value: [19, 86]})
/external/tensorflow/tensorflow/contrib/copy_graph/python/util/
Dcopy_elements.py87 init_value = temp_session.run(org_instance.initialized_value())
92 init_value,
/external/llvm/test/CodeGen/PowerPC/
Dctrloop-fp64.ll29 @init_value = global double 1.000000e+00, align 8
34 %0 = load double, double* @init_value, align 8
/external/pcre/dist2/src/sljit/
DsljitNativePPC_64.c392 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 reg, sljit_sw init_value) in emit_const() argument
394 FAIL_IF(push_inst(compiler, ADDIS | D(reg) | A(0) | IMM(init_value >> 48))); in emit_const()
395 FAIL_IF(push_inst(compiler, ORI | S(reg) | A(reg) | IMM(init_value >> 32))); in emit_const()
397 FAIL_IF(push_inst(compiler, ORIS | S(reg) | A(reg) | IMM(init_value >> 16))); in emit_const()
398 return push_inst(compiler, ORI | S(reg) | A(reg) | IMM(init_value)); in emit_const()
DsljitNativeSPARC_32.c142 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const() argument
144 FAIL_IF(push_inst(compiler, SETHI | D(dst) | ((init_value >> 10) & 0x3fffff), DR(dst))); in emit_const()
145 return push_inst(compiler, OR | D(dst) | S1(dst) | IMM_ARG | (init_value & 0x3ff), DR(dst)); in emit_const()
DsljitNativePPC_32.c247 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 reg, sljit_sw init_value) in emit_const() argument
249 FAIL_IF(push_inst(compiler, ADDIS | D(reg) | A(0) | IMM(init_value >> 16))); in emit_const()
250 return push_inst(compiler, ORI | S(reg) | A(reg) | IMM(init_value)); in emit_const()
DsljitNativeMIPS_64.c439 …IT_INLINE sljit_s32 emit_const(struct sljit_compiler *compiler, sljit_s32 dst, sljit_sw init_value) in emit_const() argument
441 FAIL_IF(push_inst(compiler, LUI | T(dst) | IMM(init_value >> 48), DR(dst))); in emit_const()
442 FAIL_IF(push_inst(compiler, ORI | S(dst) | T(dst) | IMM(init_value >> 32), DR(dst))); in emit_const()
444 FAIL_IF(push_inst(compiler, ORI | S(dst) | T(dst) | IMM(init_value >> 16), DR(dst))); in emit_const()
446 return push_inst(compiler, ORI | S(dst) | T(dst) | IMM(init_value), DR(dst)); in emit_const()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc362 TF_ASSIGN_OR_RETURN(llvm::Value * init_value, in MakeElementGenerator()
364 ir_builder_->CreateStore(init_value, accum_ptr); in MakeElementGenerator()
422 TF_ASSIGN_OR_RETURN(llvm::Value * init_value, in MakeElementGenerator()
424 ir_builder()->CreateStore(init_value, accum_ptr); in MakeElementGenerator()
/external/tensorflow/tensorflow/compiler/xla/client/
Dcomputation_builder.cc1271 const ComputationDataHandle& init_value, const Computation& computation, in Reduce() argument
1276 *request->mutable_init_value() = init_value; in Reduce()
1286 const ComputationDataHandle& init_value, const Computation& computation) { in ReduceAll() argument
1298 return Reduce(operand, init_value, computation, all_dimnos); in ReduceAll()
1303 const ComputationDataHandle& init_value, const Computation& computation, in ReduceWindow() argument
1326 return ReduceWindowWithGeneralPadding(operand, init_value, computation, in ReduceWindow()
1333 const ComputationDataHandle& init_value, const Computation& computation, in ReduceWindowWithGeneralPadding() argument
1341 *request->mutable_init_value() = init_value; in ReduceWindowWithGeneralPadding()
1414 const ComputationDataHandle& init_value, const Computation& scatter) { in SelectAndScatter() argument
1427 source, init_value, scatter); in SelectAndScatter()
[all …]
Dcomputation_builder.h532 const ComputationDataHandle& init_value, const Computation& computation,
538 const ComputationDataHandle& init_value,
544 const ComputationDataHandle& init_value, const Computation& computation,
552 const ComputationDataHandle& init_value, const Computation& computation,
568 const ComputationDataHandle& init_value, const Computation& scatter);
578 const ComputationDataHandle& init_value, const Computation& scatter);

123