Searched defs:var_shape (Results 1 – 11 of 11) 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
242 TensorShape var_shape, accum_shape; in Compile() local
291 TensorShape var_shape, accum_shape; in Compile() local
344 TensorShape var_shape, accum_shape; in Compile() local
400 TensorShape var_shape, accum_shape, squared_accum_shape; in Compile() local
476 TensorShape var_shape, m_shape, v_shape; in Compile() local
572 TensorShape var_shape, m_shape, v_shape; in Compile() local
[all …]
|
| D | stateful_random_ops.cc | 190 TensorShape var_shape; in CompileImpl() local
437 TensorShape var_shape; in Compile() local
|
| D | variable_ops.cc | 162 TensorShape var_shape; in Compile() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 224 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
310 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/core/framework/ |
| D | variable.proto | 83 repeated int64 var_shape = 4; field
|
| /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 | 1461 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 | 1277 var_shape=None, argument
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2703 TensorShape var_shape; in CreatePerHostDummyArgs() local
|