Searched refs:zero_tensor (Results 1 – 2 of 2) sorted by relevance
393 Tensor zero_tensor(data_type, TensorShape({})); in SparsifyGatherInternal() local394 zero_tensor.flat<float>()(0) = 0.0; in SparsifyGatherInternal()396 CreateConstNode(zero_tensor, StrCat(gather_node.name(), "/Const"), in SparsifyGatherInternal()
337 Tensor zero_tensor(DT_INT32, TensorShape()); in MakeReductionAxes() local338 zero_tensor.flat<int32>()(0) = 0; in MakeReductionAxes()342 .Attr("value", zero_tensor) in MakeReductionAxes()