/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/compiler/tf2xla/kernels/ |
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/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...] |
/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...] |