Searched defs:var_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 36 TensorShape var_shape; in Compile() local
81 TensorShape var_shape; in Compile() local
126 TensorShape var_shape, accum_shape; in Compile() local
184 TensorShape var_shape, accum_shape; in Compile() local
240 TensorShape var_shape, accum_shape; in Compile() local
281 TensorShape var_shape, accum_shape; in Compile() local
337 TensorShape var_shape, accum_shape, squared_accum_shape; in Compile() local
412 TensorShape var_shape, m_shape, v_shape; in Compile() local
497 TensorShape var_shape, m_shape, v_shape; in Compile() local
568 TensorShape var_shape, ms_shape, mom_shape, mg_shape; in Compile() local
[all …]
|
| D | stateful_random_ops.cc | 198 TensorShape var_shape; in CompileImpl() local
|
| D | variable_ops.cc | 164 TensorShape var_shape; in Compile() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 42 repeated int64 var_shape = 4; field
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 241 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
318 ShapeHandle var_shape = handle_shape_and_type[0].shape; in ResourceScatterUpdateShape() local
|
| D | state_ops.cc | 117 ShapeHandle var_shape = c->input(0); in ScatterUpdateShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander_test.cc | 95 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
|
| D | shape_inference.cc | 1417 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 201 var_name, var_shape, expected_axis_shards, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 1085 var_shape=None, argument
|