| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| training_ops.cc | 36 TensorShape var_shape; variable
37 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &handle));
46 ctx, var_shape.IsSameSize(delta_shape),
48 var_shape.DebugString(), " vs ",
70 TensorShape var_shape, accum_shape; variable
72 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var));
75 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape),
78 var_shape.DebugString(), " ", accum_shape.DebugString()));
86 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape),
89 var_shape.DebugString(), " ", grad_shape.DebugString()))
128 TensorShape var_shape, accum_shape; variable
170 TensorShape var_shape, m_shape, v_shape; variable
262 TensorShape var_shape, ms_shape, mom_shape; variable
346 TensorShape var_shape, accum_shape, linear_shape; local
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| resource_variable_ops.cc | 132 ShapeHandle var_shape = (*handle_data)[0].shape; local
133 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape)
180 ShapeHandle var_shape = handle_shape_and_type.shape;
186 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape));
202 ShapeHandle var_shape = handle_shape_and_type.shape;
208 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape));
|
| state_ops.cc | 117 ShapeHandle var_shape = c->input(0); local
123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape));
127 c->set_output(0, var_shape);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| partitioned_variables_test.py | 197 var_name, var_shape, expected_axis_shards,
203 var_name, dtype=dtypes.float32, shape=var_shape)
217 var_shape=[2048],
228 var_shape=[2048, 1024],
238 var_shape=[2048, 1024],
246 var_shape=[2048, 1024],
257 var_shape=[2048, 1024],
265 var_shape=[2048, 1024],
275 var_shape=[64, 1024, 1, 3],
283 var_shape=[64, 1024, 1, 3]
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander_test.cc | 80 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); local
94 HloInstruction::CreateParameter(3, var_shape, "var"));
|
| shape_inference.h | 107 const Shape& var_shape,
|
| shape_inference.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| 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/python/ops/ |
| variables.py | [all...] |
| variable_scope.py | 623 var_shape = slice_shape[:]
628 var_shape[slice_dim] += 1
629 slice_offset[slice_dim] += var_shape[slice_dim]
640 init_shape = var_shape
643 init_shape = var_shape
645 init = array_ops.slice(initializer, var_offset, var_shape)
651 init = array_ops.slice(init, var_offset, var_shape)
672 name, shape.as_list(), var_offset, var_shape))
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| slot_creator.py | 83 slice_info.var_shape[:]))
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 632 var._save_slice_info.var_shape)
636 slice_value = slice_value.reshape(var._save_slice_info.var_shape)
|
