Home
last modified time | relevance | path

Searched refs:initial_val (Results 1 – 7 of 7) sorted by relevance

/external/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c371 int32_t initial_val = initial->i32[0]; in get_iteration() local
372 int32_t span = limit->i32[0] - initial_val; in get_iteration()
378 uint32_t initial_val = initial->u32[0]; in get_iteration() local
379 uint32_t span = limit->u32[0] - initial_val; in get_iteration()
387 float initial_val = initial->f32[0]; in get_iteration() local
388 float span = limit->f32[0] - initial_val; in get_iteration()
589 nir_const_value initial_val = in find_trip_count() local
600 int iterations = calculate_iterations(&initial_val, &step_val, in find_trip_count()
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer_test.py44 def _minimize(self, initial_val, loss_grad_func, step_callback, argument
48 loss_grad_func(initial_val)
50 step_callback(initial_val)
52 _, grad = loss_grad_func(initial_val)
53 return initial_val - grad
Dexternal_optimizer.py199 initial_val=initial_packed_var_val,
218 def _minimize(self, initial_val, loss_grad_func, equality_funcs, argument
361 def _minimize(self, initial_val, loss_grad_func, equality_funcs, argument
379 minimize_args = [loss_grad_func_wrapper, initial_val]
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc110 Node* initial_val = nullptr; // Const op in rewriteNode() local
140 .Finalize(g, &initial_val)); in rewriteNode()
144 .Input(initial_val) // value: T in rewriteNode()
Dlower_if_op_test.cc182 auto initial_val = ops::_Arg(root.WithOpName("initial_val"), DT_INT32, 1); in TEST() local
185 auto init = ops::AssignVariableOp(root.WithOpName("init"), var, initial_val); in TEST()
231 feeds.emplace(Output(initial_val.node()), Input::Initializer(10)); in TEST()
241 feeds.emplace(Output(initial_val.node()), Input::Initializer(10)); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc248 Tensor initial_val(DT_FLOAT, TensorShape({3, 7})); in TEST_F() local
249 test::FillIota<float>(&initial_val, 0); in TEST_F()
252 .Attr("value", initial_val) in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc1148 Tensor initial_val(DT_INT32, TensorShape({3})); in TEST_F() local
1149 test::FillIota<int>(&initial_val, 7); in TEST_F()
1153 .Attr("value", initial_val) in TEST_F()