Searched refs:t_out (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 340 typedef std::function<Status(const T& t, T* t_out,
361 T* t_out = to->get<T>(); in UnaryVariantDeviceCopyRegistration()
362 return device_copy_fn(t, t_out, device_copy_tensor_fn); in UnaryVariantDeviceCopyRegistration()
369 typedef std::function<Status(OpKernelContext* ctx, const T& t, T* t_out)>
389 T* t_out = v_out->get<T>(); in UnaryVariantUnaryOpRegistration()
390 return unary_op_fn(ctx, t, t_out); in UnaryVariantUnaryOpRegistration()
425 T* t_out = out->get<T>(); in UnaryVariantBinaryOpRegistration()
426 return binary_op_fn(ctx, t_a, t_b, t_out); in UnaryVariantBinaryOpRegistration()
|
| D | variant_op_registry.cc | 183 T* t_out) { in ZerosLikeVariantPrimitiveType() argument
184 *t_out = T(0); in ZerosLikeVariantPrimitiveType()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | in_topk_op.cc | 67 Tensor* t_out = nullptr; in Compute() local
70 0, TensorShape({targets_in.dim_size(0)}), &t_out)); in Compute()
71 auto out = t_out->vec<bool>(); in Compute()
|
| D | record_input_op.cc | 58 auto t_out = out.flat<string>(); in Compute() local
60 OP_REQUIRES_OK(ctx, yielder_->YieldOne(&t_out(i))); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 594 t_out.shape.most_specific_compatible_shape(f_out.shape)
595 for t_out, f_out in zip(true_graph_outputs, false_graph_outputs)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2305 TF_Tensor* t_out = nullptr; in TEST_F() local
2306 TF_OperationGetAttrTensor(oper, "v", &t_out, s_); in TEST_F()
2308 EXPECT_EQ(TF_STRING, TF_TensorType(t_out)); in TEST_F()
2309 EXPECT_EQ(0, TF_NumDims(t_out)); in TEST_F()
2310 ASSERT_EQ(TF_TensorByteSize(t_in.get()), TF_TensorByteSize(t_out)); in TEST_F()
2311 EXPECT_EQ(0, memcmp(TF_TensorData(t_in.get()), TF_TensorData(t_out), in TEST_F()
2312 TF_TensorByteSize(t_out))); in TEST_F()
2313 TF_DeleteTensor(t_out); in TEST_F()
|