Home
last modified time | relevance | path

Searched defs:input_dims (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
77 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput()
118 void SpatialConvolutionBackwardKernel(Dimensions input_dims, in SpatialConvolutionBackwardKernel()
180 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
213 void CuboidConvolutionBackwardInput(Dimensions input_dims, in CuboidConvolutionBackwardInput()
256 void CuboidConvolutionBackwardKernel(Dimensions input_dims, in CuboidConvolutionBackwardKernel()
Dreduce_join_op.cc77 int32 input_dims) { in GetReducedIndices()
92 int32 input_dims, in MakeUnreducedIndices()
127 const int32 input_dims = input_shape.dims(); in Compute() local
Dtile_ops.cc101 const int input_dims = input.dims(); in Compute() local
274 const int input_dims = input.dims(); in Compute() local
354 const std::vector<Tmultiples>& input_dims, in HandleCaseImpl()
447 OpKernelContext* context, const std::vector<Tmultiples>& input_dims, in HandleCase()
Dwhere_op.cc137 const int input_dims = input.dims(); in Compute() local
258 const int input_dims = input.dims(); in ComputeAsync() local
268 void ComputeAsyncType(const Tensor& input, const int input_dims, in ComputeAsyncType()
Dsparse_slice_op.cc58 const int input_dims = input_shape.NumElements(); in Compute() local
Deigen_benchmark_cpu_test.cc44 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolution() local
73 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardInput() local
101 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardKernel() local
266 typename Benchmark::Dimensions input_dims( in CuboidConvolution() local
296 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardInput() local
325 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardKernel() local
Dquantized_concat_op.cc118 int input_dims, const OpInputList& values, in CalculateConcatDims()
184 const int input_dims = values[0].dims(); in Compute() local
Dslice_op.cc85 const int input_dims = input.dims(); in SharedValidation() local
173 const int input_dims = input.dims(); in Compute() local
Dreverse_op.cc166 const int input_dims = input.dims(); in Compute() local
243 const int input_dims = input.dims(); in Compute() local
Dmkl_softmax_op.cc62 const int input_dims = src_tf_shape.dims(); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h39 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
66 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
82 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
98 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
108 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
143 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
163 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
186 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
203 inline void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
229 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h138 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
165 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
181 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
197 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
207 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
242 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
262 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
285 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
326 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
345 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api_test.cc46 std::array<int, 1> input_dims = {2}; in TEST() local
108 const std::array<int, 1> input_dims = {2}; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc413 std::vector<int64> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local
479 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
538 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
604 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
674 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
738 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
807 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
876 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
947 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
1022 std::vector<int64> input_dims = {1, 4, 4, 160}; in RunTest() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_reshape_transpose.cc58 std::vector<int> ComputeNewPerm(std::vector<int> input_dims, in ComputeNewPerm()
171 std::vector<int> input_dims = input_shape.dims(); in Run() local
Dconvert_trivial_transpose_to_reshape.cc78 std::vector<int> const& input_dims = input_array.shape().dims(); in Run() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc32 absl::Span<int64 const> input_dims = in BroadcastTo() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc33 auto input_dims = input_shape.dimensions(); in TopK() local
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc52 const TfLiteIntArray* input_dims = nullptr; member
141 const int input_dims = NumDimensions(input); in GetPaddedOutputShape() local
165 const int input_dims = NumDimensions(input_tensor); in Eval() local
Dsqueeze.cc47 const TfLiteIntArray* input_dims = op_context.input->dims; in Prepare() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dslice_op.cc56 const int input_dims = input_shape.dims(); in Compile() local
Dtile_ops.cc56 const int input_dims = input_shape.dims(); in Compile() local
Dindex_ops.cc48 const int input_dims = input_shape.dims(); in Compile() local
Dindex_ops_cpu.cc56 const int input_dims = input_shape.dims(); in Compile() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax_test.cc32 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestSoftmaxFloat() local
96 TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); in TestSoftmaxQuantized() local

123