/external/tensorflow/tensorflow/core/framework/ |
D | control_flow.h | 32 struct FrameAndIter { struct 36 FrameAndIter() {} in FrameAndIter() argument 38 FrameAndIter(uint64 frame, int64 iter) { in FrameAndIter() function 43 bool operator==(const FrameAndIter& other) const { 49 size_t operator()(const FrameAndIter& key) const { in operator() 51 CHECK_EQ(sizeof(uint64) + sizeof(int64), sizeof(FrameAndIter)); in operator() 52 return Hash64(reinterpret_cast<const char*>(&key), sizeof(FrameAndIter)); in operator()
|
D | rendezvous.h | 58 const FrameAndIter& frame_iter);
|
D | rendezvous_test.cc | 42 "/job:mnist/replica:1/task:2/device:GPU:0", "var0", FrameAndIter(0, 0)); in TEST() 101 name, FrameAndIter(0, 0)); in MakeKey()
|
D | rendezvous.cc | 54 const FrameAndIter& frame_iter) { in CreateKey()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | sendrecv_ops.cc | 36 const FrameAndIter& frame_iter, string* key) { in GetRendezvousKey() 42 static FrameAndIter GetFrameAndIter(OpKernelContext* ctx, in GetFrameAndIter() 49 return FrameAndIter(reinterpret_cast<uint64>(ctx->call_frame()), 0); in GetFrameAndIter() 90 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in Compute() 91 if (frame_iter == FrameAndIter(0, 0)) { in Compute() 180 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in ComputeAsync() 181 if (frame_iter == FrameAndIter(0, 0)) { in ComputeAsync()
|
D | constant_op_test.cc | 67 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
|
D | restore_op_test.cc | 173 params.frame_iter = FrameAndIter(0, 0); in TEST_F() 418 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
|
D | segment_reduction_ops_test.cc | 73 params.frame_iter = FrameAndIter(0, 0); in BM_SegmentReduction()
|
D | parse_tensor_test.cc | 57 params.frame_iter = FrameAndIter(0, 0); in ParseSerializedWithNodeDef()
|
D | ops_testutil.h | 169 params_.get()->frame_iter = FrameAndIter(0, 0); in RunOpKernel()
|
D | sparse_to_dense_op_test.cc | 252 params.frame_iter = FrameAndIter(0, 0); in BM_SparseToDense()
|
D | nn_ops_test.cc | 767 params.frame_iter = FrameAndIter(0, 0); in BM_LRNFloat() 847 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPool() 952 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPoolBk() 1042 params.frame_iter = FrameAndIter(0, 0); in BM_MaxPool() 1218 params.frame_iter = FrameAndIter(0, 0); in BM_ReluFloat()
|
D | restore_v2_op_test.cc | 182 params.frame_iter = FrameAndIter(0, 0); in RunTest()
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | rpc_rendezvous_mgr_test.cc | 90 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F() 110 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F() 144 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F() 175 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_runner.cc | 136 tensor_name, FrameAndIter(0, 0)); in Run() 194 output_names[i], FrameAndIter(0, 0)); in Run()
|
D | rendezvous_util_test.cc | 50 "/job:localhost/replica:0/task:0/device:GPU:0", name, FrameAndIter(0, 0)); in MakeStringKey()
|
D | kernel_benchmark_testlib.cc | 128 recv_device, tensor_name, FrameAndIter(0, 0)); in GetRendezvousKey()
|
D | process_function_library_runtime.cc | 120 target_device, name, FrameAndIter(0, 0)); in SendTensors() 139 target_device, name, FrameAndIter(0, 0)); in ReceiveTensorsAsync()
|
D | direct_session.cc | 137 const FrameAndIter& frame_iter) { in GetRendezvousKey() 1332 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors() 1338 FrameAndIter(0, 0)); in CreateExecutors()
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | cluster_function_library_runtime.cc | 61 target, 1 /* src_incarnation */, target, in.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph() 106 out.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | evaluation_utils.cc | 98 params.frame_iter = FrameAndIter(0, 0); in EvaluateNode()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | single_threaded_executor.cc | 230 params.frame_iter = FrameAndIter(0, 0); in RunAsync()
|
D | dataset_test_base.cc | 184 params_->frame_iter = FrameAndIter(0, 0); in CreateOpKernelContext()
|
D | single_threaded_executor_test.cc | 135 name, FrameAndIter(0, 0)), in Key()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device.cc | 233 params.frame_iter = FrameAndIter(0, 0); in Run()
|