Home
last modified time | relevance | path

Searched defs:axes_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreverse_op.cc76 const TensorShape axes_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreduce_tester.cc118 const std::array<int32_t, 1> axes_shape{ in CreateTfLiteModel() local
Dquantized_reduce_tester.cc127 const std::array<int32_t, 1> axes_shape{ in CreateTfLiteModel() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dreduce_builder.cc58 const std::vector<int> axes_shape = {1, 1, 1, axes_size}; in PopulateSubGraph() local