Searched defs:shape_vec (Results 1 – 15 of 15) sorted by relevance
266 std::vector<int64> shape_vec; in ShapeProtoToVec() local278 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec) { in VecToShapeProto()
227 std::vector<string> shape_vec; in FormatNode() local
709 std::vector<int64>& shape_vec = input_shapes[inp.first]; in input_shapes() local
47 auto shape_vec = shape_t.vec<int64>(); in Compute() local
83 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
260 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local415 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local494 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
252 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local589 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() local
349 auto shape_vec = shape_tensor.flat<int32>(); in Compute() local
174 const auto shape_vec = shape_t->vec<int64>(); in Compute() local
531 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
505 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
855 std::vector<TensorShape> shape_vec; in BuildClusterSubgraphDef() local1416 std::vector<TensorShape> shape_vec; in BuildNodeMapFromOpsDefinitions() local
95 auto shape_vec = shape_t.vec<int32>(); in Compute() local
574 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors() local
1642 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local