Searched defs:reshape_dims (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | adjust_contrast_op.h | 52 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
65 reshape_dims; in operator() local
114 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
132 reshape_dims; in operator() local
|
| D | cwise_op_gpu_select.cu.cc | 69 Eigen::Tensor<int, 2>::Dimensions reshape_dims{{ batch, 1 }}; in operator ()() local
73 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()() local
|
| D | cwise_op_select.cc | 270 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}}; in operator ()() local
274 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; in operator ()() local
|
| D | eigen_spatial_convolutions_test.cc | 1490 NewDimension reshape_dims; in PackRhsHelper() local
1626 NewDimension reshape_dims; in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_expanddims_to_reshape.cc | 58 std::vector<int> reshape_dims(input_array.shape().dims()); in Run() local
|
| D | convert_reorder_axes.cc | 40 std::vector<int> reshape_dims = {1, input_shape.dims(0), input_shape.dims(1), in CreateReshapeFromReorderAxes() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_padding.cpp | 63 Eigen::DSizes<ptrdiff_t, 2> reshape_dims; in test_padded_expr() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 264 std::vector<int64> reshape_dims(warp_shape.dims()); in CalculateGradData() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1180 nvinfer1::Dims reshape_dims; in TransposeTensor() local
2131 nvinfer1::Dims reshape_dims; in ConvertReshape() local
2361 nvinfer1::Dims reshape_dims; in ConvertStridedSliceHelper() local
2968 nvinfer1::Dims reshape_dims; in ConvertBiasAdd() local
3006 nvinfer1::Dims reshape_dims = original_dims; in ConvertBiasAdd() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 742 NodeDef reshape_dims; in QuantizeNodes() local
|