| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 36 TensorShape var_shape; in Compile() local
37 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &handle)); in Compile()
46 ctx, var_shape.IsSameSize(delta_shape), in Compile()
48 var_shape.DebugString(), " vs ", in Compile()
81 TensorShape var_shape; in Compile() local
82 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &var_shape, &var)); in Compile()
98 ctx, var_shape.IsSameSize(delta_shape), in Compile()
100 var_shape.DebugString(), " vs ", in Compile()
126 TensorShape var_shape, accum_shape; in Compile() local
128 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
[all …]
|
| D | stateful_random_ops.cc | 190 TensorShape var_shape; in CompileImpl() local
192 state_input_idx, STATE_ELEMENT_DTYPE, &var_shape, &var)); in CompileImpl()
193 TF_RETURN_IF_ERROR(CheckStateShape(alg, var_shape)); in CompileImpl()
437 TensorShape var_shape; in Compile() local
440 &var_shape, &var)); in Compile()
446 OP_REQUIRES_OK(ctx, CheckStateShape(alg, var_shape)); in Compile()
|
| D | variable_ops.cc | 162 TensorShape var_shape; in Compile() local
165 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value)); in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 201 var_name, var_shape, expected_axis_shards, argument
207 var_name, dtype=dtypes.float32, shape=var_shape)
221 var_shape=[2048],
232 var_shape=[2048, 1024],
242 var_shape=[2048, 1024],
250 var_shape=[2048, 1024],
261 var_shape=[2048, 1024],
269 var_shape=[2048, 1024],
279 var_shape=[64, 1024, 1, 3],
287 var_shape=[64, 1024, 1, 3],
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 224 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
225 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape) in VariableShapeShapeFn()
310 ShapeHandle var_shape = handle_shape_and_type[0].shape; in ResourceScatterUpdateShape() local
316 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ResourceScatterUpdateShape()
|
| D | state_ops.cc | 117 ShapeHandle var_shape = c->input(0); in ScatterUpdateShape() local
123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ScatterUpdateShape()
130 c->set_output(0, var_shape); in ScatterUpdateShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander_test.cc | 95 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
109 HloInstruction::CreateParameter(3, var_shape, "var")); in TEST_F()
|
| D | shape_inference.cc | 1461 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape() argument
1467 TF_RETURN_IF_ERROR(ExpectArray(var_shape, "var input of batch norm grad")); in InferBatchNormGradShape()
1474 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(var_shape)); in InferBatchNormGradShape()
1508 if (var_shape.rank() != 1) { in InferBatchNormGradShape()
1512 var_shape.rank()); in InferBatchNormGradShape()
1559 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(var_shape, in InferBatchNormGradShape()
1590 if (ShapeUtil::GetDimension(var_shape, 0) != feature_count) { in InferBatchNormGradShape()
1595 ShapeUtil::GetDimension(var_shape, 0), feature_count); in InferBatchNormGradShape()
|
| D | shape_inference.h | 104 const Shape& var_shape,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 1277 var_shape=None, argument
1301 self.var_shape = [i for i in save_slice_info_def.var_shape]
1306 self.var_shape = var_shape
1313 "%d,%d" % (o, s) for o, s in zip(self.var_offset, self.var_shape))
1334 for i in self.var_shape:
1335 save_slice_info_def.var_shape.append(i)
|
| D | variable_scope.py | 757 for i, (var_offset, var_shape) in enumerate(
771 init_shape = var_shape
774 init_shape = var_shape
776 init = array_ops.slice(initializer, var_offset, var_shape)
782 init = array_ops.slice(init, var_offset, var_shape)
806 var_shape))
|
| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator.py | 107 slice_info.var_shape[:n]))
|
| D | slot_creator_test.py | 160 self.assertAllEqual([1], si.var_shape)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 83 repeated int64 var_shape = 4; field
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator_test.py | 603 var_shape = tuple()
622 var._type_spec = tensor_spec.TensorSpec(var_shape, var_dtype, var_name)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 549 var_shape=v.shape.as_list())
|
| D | sharded_variable_test.py | 338 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 381 var_shape = resource_variable_ops.variable_shape(var.handle)
391 return_elements=[var.name, var_assign.name, var_shape.name,
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2703 TensorShape var_shape; in CreatePerHostDummyArgs() local
2704 if (!raw_var_shape.handle_shape.AsTensorShape(&var_shape) && in CreatePerHostDummyArgs()
2705 !raw_var_shape.shape.AsTensorShape(&var_shape)) { in CreatePerHostDummyArgs()
2717 for (int i = 0; i < var_shape.dims(); ++i) { in CreatePerHostDummyArgs()
2718 tensorshape_proto.add_int_val(var_shape.dim_size(i)); in CreatePerHostDummyArgs()
2720 TensorShape shape_shape({var_shape.dims()}); in CreatePerHostDummyArgs()
|