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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 44 TensorShape delta_shape = ctx->InputShape(2); variable
46 ctx, var_shape.IsSameSize(delta_shape),
49 delta_shape.DebugString()));
96 TensorShape delta_shape = ctx->InputShape(4); variable
98 ctx, var_shape.IsSameSize(delta_shape),
101 delta_shape.DebugString()));
    [all...]
image_ops.cc 278 const TensorShape& delta_shape = context->InputShape(1); variable
282 OP_REQUIRES(context, TensorShapeUtils::IsScalar(delta_shape),
284 delta_shape.DebugString()));

Completed in 125 milliseconds