Home
last modified time | relevance | path

Searched refs:split_dim_orig (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsplit_op.cc45 int64 split_dim_orig; in Compile() local
46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &split_dim_orig)); in Compile()
48 int32 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile()
49 : split_dim_orig; in Compile()
54 split_dim_orig)); in Compile()
66 split_dim_orig, " (size = ", input_shape.dim_size(split_dim), ") ", in Compile()
108 int64 split_dim_orig; in Compile() local
109 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(2, &split_dim_orig)); in Compile()
110 int64 split_dim = split_dim_orig < 0 ? split_dim_orig + input_shape.dims() in Compile()
111 : split_dim_orig; in Compile()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsplit_op.cc58 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases() local
60 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in ComputeEasyCases()
67 "), but got ", split_dim_orig)); in ComputeEasyCases()
216 const int32 split_dim_orig = context->input(0).flat<int32>()(0); in Compute() local
218 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in Compute()
286 const int32 split_dim_orig = context->input(0).flat<int32>()(0); in Compute() local
288 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in Compute()
343 const int32 split_dim_orig = context->input(0).flat<int32>()(0); in Compute() local
345 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in Compute()
Dsplit_v_op.cc65 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases() local
67 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in ComputeEasyCases()
93 "), but got ", split_dim_orig)); in ComputeEasyCases()
271 const int32 split_dim_orig = context->input(2).flat<int32>()(0); in Compute() local
273 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in Compute()
350 const int32 split_dim_orig = context->input(2).flat<int32>()(0); in Compute() local
352 split_dim_orig < 0 ? split_dim_orig + input.dims() : split_dim_orig; in Compute()