Searched defs:input0_array (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dresolve_constant_binary.cc70 const auto& input0_array = model->GetArray(binary_op->inputs[0]); local
74 CHECK(input0_array.data_type == InputsDataType);
80 CHECK(input0_array.buffer);
85 const auto& input0_data = input0_array.GetBuffer<InputsDataType>().data;
98 Shape input0_shape = input0_array.shape();
211 const auto& input0_array = model->GetArray(binary_op->inputs[0]); local
214 if (!input0_array.buffer || !input1_array.buffer) {
236 CHECK(input0_array.data_type == input1_array.data_type);
H A Dpropagate_fixed_sizes.cc416 const auto& input0_array = model->GetArray(op->inputs[0]); local
419 if (!input0_array.has_shape() || !input1_array.has_shape()) {
424 ComputeBinaryOperatorOutputSize(input0_array.shape(), input1_array.shape(),
445 const auto& input0_array = model->GetArray(op->inputs[0]); local
447 output_array.copy_shape(input0_array.shape());

Completed in 134 milliseconds