| /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 | 191 TensorShape var_shape; in CompileImpl() local
193 state_input_idx, STATE_ELEMENT_DTYPE, &var_shape, &var)); in CompileImpl()
194 TF_RETURN_IF_ERROR(CheckStateShape(alg, var_shape)); in CompileImpl()
438 TensorShape var_shape; in Compile() local
441 &var_shape, &var)); in Compile()
447 OP_REQUIRES_OK(ctx, CheckStateShape(alg, var_shape)); in Compile()
|
| D | variable_ops.cc | 178 TensorShape var_shape; in Compile() local
181 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value)); in Compile()
185 var_shape, context->InputShape(1), in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | partitioned_variables_test.py | 196 var_name, var_shape, expected_axis_shards, argument
202 var_name, dtype=dtypes.float32, shape=var_shape)
216 var_shape=[2048],
227 var_shape=[2048, 1024],
237 var_shape=[2048, 1024],
245 var_shape=[2048, 1024],
256 var_shape=[2048, 1024],
264 var_shape=[2048, 1024],
274 var_shape=[64, 1024, 1, 3],
282 var_shape=[64, 1024, 1, 3],
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 240 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
241 int64_t rank = c->RankKnown(var_shape) ? c->Rank(var_shape) in VariableShapeShapeFn()
326 ShapeHandle var_shape = handle_shape_and_type[0].shape; in ResourceScatterUpdateShape() local
332 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 | 94 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
108 HloInstruction::CreateParameter(3, var_shape, "var")); in TEST_F()
|
| D | shape_inference.cc | 1460 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape() argument
1466 TF_RETURN_IF_ERROR(ExpectArray(var_shape, "var input of batch norm grad")); in InferBatchNormGradShape()
1473 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(var_shape)); in InferBatchNormGradShape()
1507 if (var_shape.rank() != 1) { in InferBatchNormGradShape()
1511 var_shape.rank()); in InferBatchNormGradShape()
1558 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(var_shape, in InferBatchNormGradShape()
1589 if (ShapeUtil::GetDimension(var_shape, 0) != feature_count) { in InferBatchNormGradShape()
1594 ShapeUtil::GetDimension(var_shape, 0), feature_count); in InferBatchNormGradShape()
|
| D | shape_inference.h | 102 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.experimental.dtensor.-d-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| 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 | 1269 var_shape=None, argument
1293 self.var_shape = [i for i in save_slice_info_def.var_shape]
1298 self.var_shape = var_shape
1305 "%d,%d" % (o, s) for o, s in zip(self.var_offset, self.var_shape))
1326 for i in self.var_shape:
1327 save_slice_info_def.var_shape.append(i)
|
| D | variable_scope.py | 763 for i, (var_offset, var_shape) in enumerate(
777 init_shape = var_shape
780 init_shape = var_shape
782 init = array_ops.slice(initializer, var_offset, var_shape)
788 init = array_ops.slice(init, var_offset, var_shape)
812 var_shape))
|
| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator.py | 103 slice_info.var_shape[:n]))
|
| D | slot_creator_test.py | 157 self.assertAllEqual([1], si.var_shape)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 83 repeated int64 var_shape = 4; field
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3005 xla::StatusOr<Node*> CreateTpuExecuteDummyArg(const TensorShape& var_shape, in CreateTpuExecuteDummyArg() argument
3023 for (int i = 0; i < var_shape.dims(); ++i) { in CreateTpuExecuteDummyArg()
3024 tensorshape_proto.add_int_val(var_shape.dim_size(i)); in CreateTpuExecuteDummyArg()
3026 TensorShape shape_shape({var_shape.dims()}); in CreateTpuExecuteDummyArg()
3093 TensorShape var_shape; in CreatePartitionedDummyVarArgs() local
3094 if (!raw_shape.handle_shape.AsTensorShape(&var_shape) && in CreatePartitionedDummyVarArgs()
3095 !raw_shape.shape.AsTensorShape(&var_shape)) { in CreatePartitionedDummyVarArgs()
3098 TF_RETURN_IF_ERROR(ComputeShardedArgShapes(&var_shape, sharding)); in CreatePartitionedDummyVarArgs()
3114 CreateTpuExecuteDummyArg(var_shape, dtype, host_device, in CreatePartitionedDummyVarArgs()
3217 TensorShape var_shape; in CreateOrGetPerHostVariableCopy() local
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 377 var_shape = resource_variable_ops.variable_shape(var.handle)
387 return_elements=[var.name, var_assign.name, var_shape.name,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 719 var_shape=v.shape.as_list())
|
| D | sharded_variable_test.py | 457 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator_test.py | 680 var_shape = tuple()
699 var._type_spec = tensor_spec.TensorSpec(var_shape, var_dtype, var_name)
|