| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| dynamic_stitch_op.cc | 69 for (int input_num = 0; input_num < indices_input.size(); input_num++) {
72 XLAShapeToTensorShape(indices_input[input_num].shape(),
74 const TensorShape& data_shape = data_shapes[input_num];
77 errors::InvalidArgument("data[", input_num,
79 " does not start with indices[", input_num,
83 input_num == 0 || SameExtraShape(data0_shape, indices0_shape,
87 input_num, "].shape[", indices_shape.dims(),
89 input_num, "].shape = ", data_shape.DebugString()
[all...] |
| while_op.cc | 239 int input_num = body.input_mapping[i]; local
240 if (ctx->input_type(input_num) == DT_RESOURCE) {
242 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.cc | 104 int input_num = 0; local
106 while (input_num < num_constant_args) {
107 signature->arg_values[input_num] = ctx->input(input_num);
108 ++input_num;
111 while (input_num < ctx->num_inputs() - variable_args.size()) {
112 signature->arg_types.emplace_back(ctx->input_dtype(input_num),
113 ctx->input(input_num).shape());
114 ++input_num;
119 TF_RET_CHECK(input_num < ctx->num_inputs())
141 int input_num = 0; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| dynamic_stitch_op.cc | 104 for (int input_num = 0; input_num < indices_inputs->size(); input_num++) {
105 const Tensor& indices = (*indices_inputs)[input_num];
106 const Tensor& data = (*data_inputs)[input_num];
109 errors::InvalidArgument("data[", input_num,
111 " does not start with indices[", input_num,
114 c, input_num == 0 || SameExtraShape(data0, indices0, data, indices),
116 "Need data[0].shape[", indices0.dims(), ":] = data[", input_num,
119 ", data[", input_num, "].shape = ", data.shape().DebugString()
296 OnInputNumber(input_num); variable
[all...] |
| summary_op.cc | 137 for (int input_num = 0; input_num < c->num_inputs(); input_num++) {
138 const Tensor& in = c->input(input_num);
|
| /frameworks/compile/mclinker/tools/mcld/ |
| Main.cpp | 821 size_t input_num = 0; local
849 ++input_num;
877 ++input_num;
966 ++input_num;
975 if (input_num == 0) {
[all...] |
