Searched refs:input_vars (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 310 std::vector<mlir::Value> input_vars; in CreateNaiveMlirConv() local
311 input_vars.push_back(loop_n.getInductionVar()); in CreateNaiveMlirConv()
312 input_vars.push_back(loop_c.getInductionVar()); in CreateNaiveMlirConv()
313 input_vars.insert(input_vars.end(), output_spatial_indvars.begin(), in CreateNaiveMlirConv()
315 input_vars.insert(input_vars.end(), filter_spatial_indvars.begin(), in CreateNaiveMlirConv()
326 input_vars), in CreateNaiveMlirConv()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops_test.py | 409 input_vars = [
416 tape.watch(input_vars)
417 addn = math_ops.add_n(input_vars)
418 add_n_grad = tape.gradient(addn, input_vars)
420 addn = math_ops.add_n(input_vars)
421 add_n_grad = gradients.gradients(addn, input_vars)
539 input_vars = [a, b]
545 tape.watch(input_vars)
546 c_grad0 = tape.gradient(math_ops.divide(a, b), input_vars)
548 tape.watch(input_vars)
[all …]
|
| D | special_math_ops_test.py | 1115 input_vars = []
1118 input_vars.append(
1122 if len(input_vars) <= 2:
1125 special_math_ops.einsum(equation, *input_vars),
1133 equation, *input_vars, optimize=optimize),
|
| D | gradients_test.py | 1509 def TestFn(inputs, input_vars): argument
1510 return inputs * input_vars
1512 def TestFnSeq(inputs, input_vars): argument
1513 return (inputs * input_vars, inputs * input_vars * 2.0)
|
| D | control_flow_ops.py | 541 def _SetShapeInvariants(input_vars, enter_vars, shapes): argument
560 for inp, var, shape in zip(input_vars, enter_vars, flat_shapes):
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | einsum_op_test.py | 429 input_vars = []
432 input_vars.append(
439 special_math_ops.einsum(equation, *input_vars),
446 gen_linalg_ops.einsum(input_vars, equation),
|
| /external/mesa3d/src/mesa/program/ |
| D | prog_to_nir.c | 53 nir_variable *input_vars[VARYING_SLOT_MAX]; member
142 nir_variable *var = c->input_vars[prog_src->Index]; in ptn_get_src()
879 c->input_vars[i] = fullvar; in setup_registers_and_variables()
884 c->input_vars[i] = var; in setup_registers_and_variables()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_linking_helpers.c | 840 nir_variable *input_vars[MAX_VARYING] = { 0 }; in nir_link_xfb_varyings() local
847 input_vars[location] = var; in nir_link_xfb_varyings()
859 if (input_vars[location]) { in nir_link_xfb_varyings()
860 input_vars[location]->data.always_active_io = true; in nir_link_xfb_varyings()
|