HomeSort by relevance Sort by last modified time
    Searched refs:GetTensorShape (Results 1 - 15 of 15) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
exp_test.cc 45 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
reshape_test.cc 43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
space_to_depth_test.cc 46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
squeeze_test.cc 56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
batch_to_space_nd_test.cc 42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
mean_test.cc 40 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
pad_test.cc 37 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
split_test.cc 58 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); }
gather_test.cc 62 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
space_to_batch_nd_test.cc 42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
test_util.h 189 std::vector<int> GetTensorShape(int index) {
transpose_test.cc 139 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
strided_slice_test.cc 62 std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_tfgraph_builder.cc 50 TensorShapeProto GetTensorShape(const HloInstruction* instruction) {
157 *shapes.mutable_list()->add_shape() = GetTensorShape(instruction);
  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_nodes.cc 73 inline nvinfer1::Dims GetTensorShape(const tensorflow::Tensor& tensor) {
    [all...]

Completed in 123 milliseconds