| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A 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 …]
|
| H A D | stateful_random_ops.cc | 191 TensorShape var_shape; in CompileImpl() local
438 TensorShape var_shape; in Compile() local
|
| H A D | variable_ops.cc | 178 TensorShape var_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | resource_variable_ops.cc | 240 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
326 ShapeHandle var_shape = handle_shape_and_type[0].shape; in ResourceScatterUpdateShape() local
|
| H A D | state_ops.cc | 117 ShapeHandle var_shape = c->input(0); in ScatterUpdateShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | variable.proto | 83 repeated int64 var_shape = 4; field
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/ |
| D | variable.pb.h | 1083 inline ::int64_t SaveSliceInfoDef::var_shape(int index) const { in var_shape() function
1103 SaveSliceInfoDef::var_shape() const { in var_shape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | batchnorm_expander_test.cc | 94 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
|
| H A D | shape_inference.cc | 1460 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| H A D | partitioned_variables_test.py | 196 var_name, var_shape, expected_axis_shards, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 3005 xla::StatusOr<Node*> CreateTpuExecuteDummyArg(const TensorShape& var_shape, in CreateTpuExecuteDummyArg()
3093 TensorShape var_shape; in CreatePartitionedDummyVarArgs() local
3217 TensorShape var_shape; in CreateOrGetPerHostVariableCopy() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | variables.py | 1269 var_shape=None, argument
|