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

  /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));
127 c->set_output(0, var_shape);
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));
  /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"));
  /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...]

Completed in 1367 milliseconds