Home
last modified time | relevance | path

Searched defs:shape_vector (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/
H A Dsaved_variable_loading_test.cc116 std::vector<int64_t> shape_vector = std::get<1>(test_params); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc378 const auto& shape_vector = global_shape(); in DebugString() local
482 const auto& shape_vector = global_shape(); in DebugString() local
H A Ddtensor_device_util.h516 std::string ShapeToDebugString(const std::vector<T> shape_vector) { in ShapeToDebugString()
H A Ddtensor_device.cc1113 gtl::InlinedVector<int64, 4> shape_vector( in UpdateOutputLayoutsWithSameShapePolicy() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcount_ops.cc231 const auto shape_vector = shape.vec<int64_t>(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc301 auto shape_vector = output_type.getShape(); in GetShape() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp454 std::vector<c10::ShapeSymbol> shape_vector = shape.sizes().value(); in ComputeShapeFromReshape() local