Searched defs:shape_vec (Results 1 – 14 of 14) 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
708 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
277 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local435 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local521 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute() local
96 auto shape_vec = shape_t.vec<int32>(); in Compute() local
185 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
2325 auto shape_vec = shape_tensor->flat<int64>(); in SparseReduceShapeFn() local