Searched refs:input_tensor_size (Results 1 – 6 of 6) sorted by relevance
626 operation->input_tensor_size = operation->input_width * in etna_ml_lower_transpose()653 operation->input_tensor_size = operation->input_width * in etna_ml_lower_detranspose()682 operation->input_tensor_size = operation->input_width * in etna_ml_lower_reshuffle()
61 unsigned input_tensor_size; member
166 etna_ml_create_tensor(subgraph, operation->input_tensor, operation->input_tensor_size); in lower_operations()172 operation->input_tensor_size / 2); in lower_operations()
417 operation->input_tensor_size = operation->input_width * in etna_ml_lower_convolution()483 operation->input_tensor_size = operation->input_width * in etna_ml_lower_add()
176 uvec4 input_tensor_size; in binary_op_tensor() member344 uvec4 input_tensor_size; in binary_op_tensor_() member
1588 std::vector<int64_t> input_tensor_size; in CalculateInputTensorSize() local1589 input_tensor_size.reserve(op_info.inputs().size()); in CalculateInputTensorSize()1591 input_tensor_size.push_back( in CalculateInputTensorSize()1594 return input_tensor_size; in CalculateInputTensorSize()