Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtranspose_test.cc210 absl::Span<int64_t const> dims_out, in TransposeUsingEigenNd()
237 absl::Span<int64_t const> dims_out, in TransposeUsingEigen()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Deinsum.cc530 nvinfer1::Dims dims_out = {output_rank, {}}; in ShuffleEinsumOutput() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc94 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc124 const std::vector<int>& dims_out) { in DescribeSparseTensorDenseMatMul()