Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc55 size_t shape_dims = 1 + (data[data_ix++] % kMaxShapeDims); in FuzzImpl() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc105 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc613 auto* shape_dims = shape->mutable_dims(); in ExtendShape() local
624 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc139 const std::vector<int32> shape_dims = {static_cast<int32>(vals.size())}; in MakeConstNodeDef() local