Searched refs:shape_vector (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 116 std::vector<int64_t> shape_vector = std::get<1>(test_params); in TEST_P() local
117 TensorShape shape(shape_vector); in TEST_P()
127 testing::CreateTensorHandle(context(), dtype, shape_vector, 42); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | count_ops.cc | 231 const auto shape_vector = shape.vec<int64_t>(); in Compute() local
238 indices_values(i, j) >= 0 && indices_values(i, j) < shape_vector(j), in Compute()
241 indices_values(i, j), " which is not in [0, ", shape_vector(j), in Compute()
265 int num_batches = is_1d ? 1 : shape_vector(0); in Compute()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | shape_type_inference.cpp | 454 std::vector<c10::ShapeSymbol> shape_vector = shape.sizes().value(); in ComputeShapeFromReshape() local
456 !input_shape_vector.empty() || !shape_vector.empty(), in ComputeShapeFromReshape()
458 if (shape_vector.empty()) { in ComputeShapeFromReshape()
466 auto it_0 = std::find_if(shape_vector.begin(), shape_vector.end(), is_zero); in ComputeShapeFromReshape()
467 bool shape_has_zero = it_0 != shape_vector.end(); in ComputeShapeFromReshape()
470 for (auto i : c10::irange(shape_vector.size())) { in ComputeShapeFromReshape()
471 if (shape_vector[i].value() == -1) { in ComputeShapeFromReshape()
508 int shape_size = static_cast<int>(shape_vector.size()); in ComputeShapeFromReshape()
513 c10::ShapeSymbol& target_shape = shape_vector[i]; in ComputeShapeFromReshape()
543 shape_vector[i].value() == 0 ? input_shape_vector[i] : shape_vector[i]); in ComputeShapeFromReshape()
[all …]
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device_util.h | 516 std::string ShapeToDebugString(const std::vector<T> shape_vector) { in ShapeToDebugString() argument
517 std::vector<tensorflow::int64> cast_shape(shape_vector.begin(), in ShapeToDebugString()
518 shape_vector.end()); in ShapeToDebugString()
|
| D | dtensor_device_util.cc | 378 const auto& shape_vector = global_shape(); in DebugString() local
380 ", shape=", ShapeToDebugString(shape_vector), in DebugString()
482 const auto& shape_vector = global_shape(); in DebugString() local
484 ", shape=", ShapeToDebugString(shape_vector), in DebugString()
|
| D | dtensor_device.cc | 1113 gtl::InlinedVector<int64, 4> shape_vector( in UpdateOutputLayoutsWithSameShapePolicy() local
1116 shape_layout_cache_.find(FingerprintShape(shape_vector)); in UpdateOutputLayoutsWithSameShapePolicy()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 301 auto shape_vector = output_type.getShape(); in GetShape() local
303 shape.reserve(shape_vector.size()); in GetShape()
304 for (auto shape_object : shape_vector) { in GetShape()
|