HomeSort by relevance Sort by last modified time
    Searched refs:var_shape (Results 1 - 15 of 15) sorted by null

  /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 ",
81 TensorShape var_shape; variable
82 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &var_shape, &var));
98 ctx, var_shape.IsSameSize(delta_shape),
100 var_shape.DebugString(), " vs ",
126 TensorShape var_shape, accum_shape; variable
128 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var))
184 TensorShape var_shape, accum_shape; variable
240 TensorShape var_shape, accum_shape; variable
281 TensorShape var_shape, accum_shape; variable
337 TensorShape var_shape, accum_shape, squared_accum_shape; variable
412 TensorShape var_shape, m_shape, v_shape; variable
497 TensorShape var_shape, m_shape, v_shape; variable
568 TensorShape var_shape, ms_shape, mom_shape, mg_shape; variable
678 TensorShape var_shape, accum_shape, linear_shape; local
817 TensorShape var_shape, accum_shape, accum_update_shape; variable
883 TensorShape var_shape, m_shape; variable
    [all...]
stateful_random_ops.cc 198 TensorShape var_shape; local
200 state_input_idx, STATE_ELEMENT_DTYPE, &var_shape, &var));
201 if (var_shape.dims() != 1) {
204 var_shape.dims());
206 auto state_size = var_shape.dim_size(0);
variable_ops.cc 164 TensorShape var_shape; variable
167 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value));
  /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/compiler/xla/service/
batchnorm_expander_test.cc 95 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); local
109 HloInstruction::CreateParameter(3, var_shape, "var"));
shape_inference.h 104 const Shape& var_shape,
shape_inference.cc     [all...]
  /external/tensorflow/tensorflow/core/ops/
state_ops.cc 117 ShapeHandle var_shape = c->input(0); local
123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape));
130 c->set_output(0, var_shape);
resource_variable_ops.cc 241 ShapeHandle var_shape = (*handle_data)[0].shape; local
242 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape)
318 ShapeHandle var_shape = handle_shape_and_type[0].shape; local
324 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape));
  /external/tensorflow/tensorflow/python/kernel_tests/
partitioned_variables_test.py 201 var_name, var_shape, expected_axis_shards,
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/python/training/
slot_creator.py 92 slice_info.var_shape[:]))
  /external/tensorflow/tensorflow/python/ops/
variables.py     [all...]
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)
    [all...]
  /external/tensorflow/tensorflow/contrib/framework/python/ops/
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/
importer_test.py 380 var_shape = resource_variable_ops.variable_shape(var.handle)
390 return_elements=[var.name, var_assign.name, var_shape.name,
    [all...]

Completed in 770 milliseconds