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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
split_op.cc 45 int64 split_dim_orig; variable
46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &split_dim_orig));
48 int32 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims()
49 : split_dim_orig;
54 split_dim_orig));
66 split_dim_orig, " (size = ", input_shape.dim_size(split_dim), ") ",
108 int64 split_dim_orig; variable
109 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(2, &split_dim_orig));
110 int64 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims(
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
split_op.cc 58 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); local
60 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig;
67 "), but got ", split_dim_orig));
216 const int32 split_dim_orig = context->input(0).flat<int32>()(0); variable
218 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig;
286 const int32 split_dim_orig = context->input(0).flat<int32>()(0)
    [all...]
split_v_op.cc 65 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); local
67 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig;
93 "), but got ", split_dim_orig));
271 const int32 split_dim_orig = context->input(2).flat<int32>()(0); variable
273 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig;
350 const int32 split_dim_orig = context->input(2).flat<int32>()(0)
    [all...]

Completed in 2795 milliseconds