Searched refs:path_tensor (Results 1 – 5 of 5) sorted by relevance
71 const Tensor& path_tensor = context->input(0); in Compute() local74 const int32 num_data = static_cast<int32>(path_tensor.shape().dim_size(0)); in Compute()76 static_cast<int32>(path_tensor.shape().dim_size(1)); in Compute()96 const auto path = path_tensor.tensor<int32, 2>(); in Compute()
124 const Tensor& path_tensor = context->input(4); in Compute() local175 const auto path = path_tensor.tensor<int32, 2>(); in Compute()
321 def _file_path_value(self, path_tensor): argument323 if not isinstance(path_tensor, ops.Tensor):325 if path_tensor.op.type != "Const":327 if path_tensor.dtype != dtypes.string:329 str_value = path_tensor.op.get_attr("value").string_val
141 path_tensor = op.outputs[1]151 path_tensor,159 dl_du = array_ops.expand_dims(unpack_path_op(path_tensor, routing_grad), 2)
709 def _asset_path_from_tensor(path_tensor): argument721 if not isinstance(path_tensor, ops.Tensor):723 if path_tensor.op.type != "Const":725 if path_tensor.dtype != dtypes.string:727 str_values = path_tensor.op.get_attr("value").string_val