| /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 | 198 TensorShape var_shape; in CompileImpl() local
200 state_input_idx, STATE_ELEMENT_DTYPE, &var_shape, &var)); in CompileImpl()
201 if (var_shape.dims() != 1) { in CompileImpl()
204 var_shape.dims()); in CompileImpl()
206 auto state_size = var_shape.dim_size(0); in CompileImpl()
|
| D | variable_ops.cc | 164 TensorShape var_shape; in Compile() local
167 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/contrib/slim/python/slim/data/ |
| D | prefetch_queue_test.py | 167 var_shape = [None, 2]
169 p1.set_shape(var_shape)
171 p2.set_shape(var_shape)
173 p3.set_shape(var_shape)
183 capacity=10, dtypes=[dtypes.int32], shapes=[var_shape])
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 241 ShapeHandle var_shape = (*handle_data)[0].shape; in VariableShapeShapeFn() local
242 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape) in VariableShapeShapeFn()
318 ShapeHandle var_shape = handle_shape_and_type[0].shape; in ResourceScatterUpdateShape() local
324 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 | 1417 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape() argument
1423 TF_RETURN_IF_ERROR(ExpectArray(var_shape, "var input of batch norm grad")); in InferBatchNormGradShape()
1430 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(var_shape)); in InferBatchNormGradShape()
1464 if (var_shape.rank() != 1) { in InferBatchNormGradShape()
1468 var_shape.rank()); in InferBatchNormGradShape()
1515 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(var_shape, in InferBatchNormGradShape()
1546 if (ShapeUtil::GetDimension(var_shape, 0) != feature_count) { in InferBatchNormGradShape()
1551 ShapeUtil::GetDimension(var_shape, 0), feature_count); in InferBatchNormGradShape()
|
| D | shape_inference.h | 104 const Shape& var_shape,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 1085 var_shape=None, argument
1110 self.var_shape = [i for i in save_slice_info_def.var_shape]
1115 self.var_shape = var_shape
1122 "%d,%d" % (o, s) for o, s in zip(self.var_offset, self.var_shape)
1145 for i in self.var_shape:
1146 save_slice_info_def.var_shape.append(i)
|
| D | variable_scope.py | 737 for i, (var_offset, var_shape) in enumerate(_iter_slices(
753 init_shape = var_shape
756 init_shape = var_shape
758 init = array_ops.slice(initializer, var_offset, var_shape)
764 init = array_ops.slice(init, var_offset, var_shape)
787 name, shape.as_list(), var_offset, 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/core/framework/ |
| D | variable.proto | 42 repeated int64 var_shape = 4; field
|
| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator.py | 92 slice_info.var_shape[:]))
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | variables.py | 691 var._save_slice_info.var_shape)
695 slice_value = slice_value.reshape(var._save_slice_info.var_shape)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 380 var_shape = resource_variable_ops.variable_shape(var.handle)
390 return_elements=[var.name, var_assign.name, var_shape.name,
|