Home
last modified time | relevance | path

Searched refs:input_data_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py97 input_data_shape = list(input_shape)
98 for i, e in enumerate(input_data_shape):
100 input_data_shape[i] = np.random.randint(1, 4)
101 input_data = 10 * np.random.random(input_data_shape)
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc942 const auto& input_data_shape = model->GetArray(op->inputs[0]).shape(); in ProcessResizeBilinearOperator() local
957 .copy_shape(Shape({input_data_shape.dims(0), output_shape[0], in ProcessResizeBilinearOperator()
958 output_shape[1], input_data_shape.dims(3)})); in ProcessResizeBilinearOperator()
970 const auto& input_data_shape = model->GetArray(op->inputs[0]).shape(); in ProcessResizeNearestNeighborOperator() local
985 .copy_shape(Shape({input_data_shape.dims(0), output_shape[0], in ProcessResizeNearestNeighborOperator()
986 output_shape[1], input_data_shape.dims(3)})); in ProcessResizeNearestNeighborOperator()